#include "unet_v1.h"
#include "stream_tools.h"
#include "unet_v1_params.h"

void expand_block(data_stream<CONV1_2_PCH * CONV_A_BIT>& in, data_stream<O_BIT>& out)
{
    #pragma HLS INTERFACE axis register both port = in
    #pragma HLS INTERFACE axis register both port = out
    #pragma HLS INTERFACE ap_ctrl_none port = return

    #pragma HLS DATAFLOW
    expand_width<CONV1_2_PCH * CONV_A_BIT, O_BIT, CONV1_2_NOH * CONV1_2_NOW * CONV1_2_NOCH / CONV1_2_PCH>(
        in, out);

}